Memory leak on assignment 1 fixed

Other minor comment changes