Finished ex1.py