fixed some boundary issues, expolored expectations and made sure probs were right