You have to convert the prob input (and all the other inputs) into int as it is considered a string by default: prob = int(input("...")).
prob
int
string
prob = int(input("..."))
You can also verify this by simply printing type(prob).
type(prob)
548k questions
547k answers
4 comments
86.3k users