add predict usage, switch to using large dataset