Finished final HW assignment