Homework 3 Code