Get log out working