Add test for session