Cleaning up exercise 2 file