Updated README and fixed imports