codeeval 141 python