Add Edit functionality to mockup