fixing bug
lab5