Added hook for simple mode