Remove unused import
Refactor usage of mgo