Good tip. Im gonna include it in the following code, because it would brings changes in 'delete' action as well.