https://remp3indir.net/python-numpy-practice-problems/