Various fixes ahead of release

Just some miscellaneous fixes while doing some more thorough testing.