removing unnessary import in notebook