Lab3 fix