Adding new class on controller