3rd session of python exercise