Finish Assignment 2 basic task