Update README file
Small refactoring