slight updates to Lab 4