adds correct button class