FIX: set mode buttons as primary class