Update README
Minor code changes