Mini fixes and cleanups