Updating mockup to include new UI elements.