Set update action
Separate into views and add flash messages