Editing README.md for context of code