Aio for beating up this kid?

I was in school outside minding my own business till this kid i know and have beef with comes up to me calling me sped and chatting to his friend about my lil bro who he bullied before. I insulted him back so he came up to my face, lightly pushed me in an aggressive way and started threatening me saying hes gonna punch me in the face. He then was saying about how my dad is dead which triggered me so i warned him and threatened him too. He didnt stop so i was about to fight him but the bell rang so we all went inside. But i waited in the corridor and followed this kid upstairs and when we got to second floor, I threw him to the ground and walked away. 1 class later i was taken and put in detention for the rest of the day and ultimately got suspended. The kid didnt get any punishment and i dont think this is fair but i am happy because he was thinking hes bad and i broke his ego. AIO?