Python deliberate practice : project 1