updated README.md and cleaned code