Make HW4 code significantly more awesome.