Minor fixes based on feedback from @markstory