Improved visual appearance for homework 4 submission