Still Working on MainController for UI