modify some code katas