Changed some features to the edit method.