Lab 5 minor fix