Smart Interview Basic Hackerrank Python