Nearly done on assginment 2.