Add button class and play button