Adding cost kwargs to propagate_control_network plus bug fix (#460)

* initial commit

* cleaning up prints

* forgot to remove part of print

* little bug fix

* addressing comments

* fixing tests I

* fixing tests II

* buffing up tests