fixed error, basically done with lab 3