update to lab 4 structure