Adding per plug prediction.