Made some slight bug fixes.