Update to lab2