Edit feature added