Codewars python first question