Problem #6: small improvement