Log out bug fixes and some styling