Rushil updated the edit function is working