Added UI to main