added code for only conditional prob pediction