Add simple prediction plot