Small improvement to Problem 13