got probability working