need to display plots
add vectorized back propagation