Small touch-up on problem-3.clj