add (ugly, brute force) solution for problem four