Solution for the code question on the initial Zipfian Academy application