Add nicer style for editor