final code tweaks; make plots even prettier; clean up some tex