code faster

add enable_py and plot_img.