Styling fix, button fix