Fix the edit controller, use partials.