Completed the assignment 1