Updated README and added code