Interview Question exercises in Python