Adding functionality to Update Action