small if fix. lab5 is done