clean up

* remove redundant argument and tidy print statements