Improve look of button = change class