add missing button class