now can generate correct predictions