minor fixes to lab solution