Adding the class for the custom UIButton. Not yet completed.