fixed one bug in lab4