Lab 4 changes