Added UI elements by Isaac Nichols