import numpy as np so consistent across functions