Clean up buttons