Use a base class so we can re-use the buttons more easily