Add and or and xor to main_stage