Lab 5: minor changes