Create solution for Intro to python Exercise 2