Updated the completion of Assignment 2