Add more tests to session tests