Completion of assignment 1