Assignment-2---Python-Function