Added code to remove the coverage for the main class and UI controller.