Add logout to improved_login