updated README and remove unused imports