Add ColorButton class to reduce code duplication