Edit feature added and tests passed