diff --git a/ass1-doc/src/main/java/dst/ass1/doc/impl/DocumentQuery.java b/ass1-doc/src/main/java/dst/ass1/doc/impl/DocumentQuery.java
index db09bf5..9ebf699 100644
--- a/ass1-doc/src/main/java/dst/ass1/doc/impl/DocumentQuery.java
+++ b/ass1-doc/src/main/java/dst/ass1/doc/impl/DocumentQuery.java
@@ -1,12 +1,15 @@
 package dst.ass1.doc.impl;
 
-import com.mongodb.MongoClient;
+import com.mongodb.client.MongoCollection;
 import com.mongodb.client.MongoDatabase;
+import com.mongodb.client.model.Filters;
 import dst.ass1.doc.IDocumentQuery;
 import org.bson.Document;
 
 import java.util.List;
 
+import static dst.ass1.jpa.util.Constants.*;
+
 public class DocumentQuery implements IDocumentQuery {
     private final MongoDatabase db;
 
@@ -16,7 +19,8 @@ public class DocumentQuery implements IDocumentQuery {
 
     @Override
     public Document findLocationById(Long locationId) {
-        return null;
+        MongoCollection<Document> collection = db.getCollection(COLL_LOCATION_DATA);
+        return collection.find(Filters.eq(I_LOCATION, locationId)).first();
     }
 
     @Override