Done assignment 1