Added a configurable button and improved the styling