starting move to using numpy arrays