codewars python first question