Hackerrank Python Answer