added code that was turned in for hw2