lab5 solution fix