imporve and got session tests working