Clean up edit-button js