Added buttons and styling