Added more tests for the session module