Completed the coding for Assignment 2 and submitting