Fix Lab6 errors