Fixed lab 2 nearly done