Made prediction based off of expectation instead of highest probability