Finish the Assignment 1