Added styles and improved button functionality