Added UI methods to Main