Fix button
Add styling for buttons