Add buttons to OI class