hw5- minor tweaks