diff --git a/.gitignore b/.gitignore
index 4154e68..a4d22bb 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,4 +3,5 @@ planet-latest_geonames.tsv
 
 *.o
 
-*/.vscode
\ No newline at end of file
+*/.vscode/
+.vscode/
\ No newline at end of file
diff --git a/A1/id_query_naive b/A1/id_query_naive
index d12eb4a..86e67b9 100755
Binary files a/A1/id_query_naive and b/A1/id_query_naive differ
diff --git a/A1/id_query_naive.c b/A1/id_query_naive.c
index 5c5cc05..cbc6ee7 100644
--- a/A1/id_query_naive.c
+++ b/A1/id_query_naive.c
@@ -27,7 +27,14 @@ void free_naive(struct naive_data* data) {
 }
 
 const struct record* lookup_naive(struct naive_data *data, int64_t needle) {
-    return &data->rs[needle];
+    int n = data->n;
+    for (int i = 0 ; i < n ; i++) {
+        struct record *rec = &data->rs[i];
+        if (rec->osm_id == needle) {
+            return rec;
+        }
+    }
+    return NULL;
 }
 
 int main(int argc, char** argv) {