Make demo prettier and more usable + minor fixes to README

Test Plan:
None