fixing final bug in assign5