Added model for partition diagram