Add ability to log out and test