add python answer 1-2