Added session storage and initial tests