Add editing feature