worked on homework 2