Kata 3 - clean up