remove old visualize, add rmsprop found online