Basic Interview Question In Python