pull out probability calculations into seperate method