Updating README.md and fixing a few bugs in the SET_CONTEXT function