add probability etc