Update README
Small refactor