Go Basic Hackerrank Python Answer