Add some hook for disconnect