Smart Interview Basic Hackerrank Python Answer