Update to lab 3