Lab 5 fix