Finished assignment 2