exercise 2 completed
small changes made