Add data wrangling with mongodb (python)