Improving views
Updating flash message on update
Improving views