Styling buttons and fixing JS