Add probability calculator