Minor refactoring
Improved readme.md