Done assignment2