Collapsed ui classes into controller