CT lab2 fixes