Add 2D repr of hui's model