Misc. bug fixes, code cleanup