added predictions function