changed to use animation

slight performance increase using matplotlib.animate.FuncAnimate()
instead of directly calling imshow each time