Minor updates/bug fixes