Adapted to new library etc, but it just works
Got rid of cost function reporting