cleaned up code and readme to make it easier to run