Finished task 1 of assignment 2