Edit: new button class working