Refactored to add buttons that do something in JavaScript