Reproduced one of previous model figures in demo script