All message types now have their own class

Tests have been also updated accordingly.