Clean up button driver