Updated button class.