working on initial probability calculator