Hero code cleanup, added ability to screenshot, updated readme