Dividing flat controller on two classes