add button class
some changes