Lab 2 - [ 4 ] fixes