adding some simpler visualizations of the trajectory. Going to play around with hyperopt to optimize various parameters