Code Wars solution for 6, 7, and 8