Finish optional task 1 of assignment 3