Updates to mockup UI