Add controller and views for edit function