Added UI to Code