add simulation code for new paper idea
better visualizations