started using numpy for vector math