Adding support for post fit plots