Add basic session management