Add some info in readme.md and clearing up the code.