minor fixes to lab 5