Smart Interviews Basic Hackerrank Python