lab2 fixes