added more use of numpy
adding numpy stuff