Add session maintainence - hw2 part 3