Use new class in controller