Assignment2 Solution

Function update for submission