adding probability argument and docs