made edits to put it to work
small adjustments