add possibility to choose dofs to plot in the main