Small fixes to the UI. Added the CustomButtonUI class.