add initial model plotting