https://tutorial.eyehunts.com/python/how-to-take-integer-array-input-in-python-example-code/