adding todo for probability