got rid of zip() in the get_cost method.. this is slow compared to using numpys iterators.. code runs 5-6x faster now