AI ex: Make successors from puzzle solving cleaner