Adjust README
Minor code cleanup