Finished assignment 1