Trying to make a wrapper class for Button