Finished homework assignment

Finished homework assignment