Updated the Assignment 2 with the necessary Code and Comments