Labb 3 fixes