Updated README and solved a minor issue