Checked code for 2D distribution, new 2D plots!