add convergence plot to demo
import numpy as np
add convergence plot to demo