Adding exercise 2 ex2.py file