- More UI elements and logic added.