Updating README and adding more comments