Fix the python side of the add/remove class functions to send custom messages