added some docs and made it easier to run