Add tests for Session