Misc enhancements/bugfixes