Smart Interviews Basic Hackerrank Python Answer