Codewars Python Answer