Lab4: small changes