Final touches on Homework 4