Lab2 fixes