Problem #9: minor improvement