fixed lab5 issue