Added test buttons to the main class.