Edit: first draft of button edit