Revised kata
Minor revisions to code