Codewars Python First Question