Lab 3 code cleanup

purely for aesthetics