Fixes #1 and improves code formatting
README.md update