CHANGE: updates from v1-ant branch, + plot_prediction for oriented stimuli