added class to controller