Create HW1_numba.py