Fixing plot functions to show you your predictions