Refactored Mongo driver for golint