General updates/bug fixes.