Problem Based On Function In Python