fixing some TODOs in fig_a plotting stuff