Add edit functionality to all views and controllers