modify prob0 satisfy the Piazza