Adding code and updating README.md