lab4: small changes
lab4:  minor changes