Updated README and code documentation