Lab 2 fixed