Renamed the CustomButton to SelectionButton and made it an inner class