Smal/Medium difficulty exercises in Python