changes on lab1 to get it to work