Tweaked with initial distributions... And also to save my work since I just realized something.