Finished the assignment (I think)