update readme with code changes and nicer headings