Add session saving