Amend to work on codewars.