Session end stuff working