Cleaned up button class.