Add class on button
Add tests on views