checkin lab4 code changes