Fix button styling