Lab4 updates