better prediction plots