First python exercise from course