Minor code tweaks (missing semicolon)