Add maintenance mode support