fixed screen problem at button class