predict plot (not working though)