Adding the flags to get plots in notebook