Add session check