Added UI to Tests