updated to the latest koans

updating to the newer koans introduces a new function here