Create HW1_numpy.py