Finishing up homework one