add visualization, gradients