Allow setting button class