added screenshot and cleaned up some of the code