Added more prediction tests