Enhanced NOX controller