Finished homework 1 I think