Make figure same size as simulation figure