fixes in lab 3