Minor change to Edit GUI