I'd rather use the more standard `import numpy as np` here.  Other than that, we can merge this.  Thanks!! Once you make that change, I'll do the merge.