Refactor

Add css class to buttons