Added code for assignment 2