Clean up button classes