adding buttons for better UX