It is possible to use `pl.ion()` for single figure but then I think sphinx was complaining. So, we resorted to this.