could not broadcast input array from shape (121) into shape (120)

happens also in the plotting script

@christine-nam we really need to fix the problems with the script as soon as possible