Added bar, attention map plot: both need work