Added editing feature and tests passing