Lab 1 minor bug fix