made a few changes and added problem 2.4 to assignment 3