Updated Solution Assignment 2

This is my submission for Assignment 2