diff --git a/work1/single.c b/work1/single.c
index 54ef6cd..254c402 100644
--- a/work1/single.c
+++ b/work1/single.c
@@ -56,6 +56,7 @@ void quite();
 
 // Linked list functions
 void add_student(char student_name[20], char student_id[10], char course_name[20], int no_of_units, int marks[MAX_NO_OF_UNITS]);
+int count_elements(STUDENT *elements);
 int count();
 
 int main(void)
@@ -200,11 +201,11 @@ void add_student(char student_name[20], char student_id[10], char course_name[20
     }
 }
 
-int count()
+int count_elements(STUDENT *elements)
 {
     int n = 1;
     STUDENT *temp;
-    temp = head;
+    temp = elements;
     if (head == NULL)
     {
         return 0;
@@ -219,6 +220,11 @@ int count()
     return n;
 }
 
+int count()
+{
+    return count_elements(head);
+}
+
 void search_student()
 {
     char search_key[NAME_SIZE];
@@ -315,8 +321,13 @@ STUDENT *find_failed_mark(int upper_mark)
     // refresh data from file first
     read_file();
 
+    printf("Count elements: %d\n", count());
+
     STUDENT *temp = head, *failed_students = NULL;
 
+    int count = count_elements(temp);
+    printf("Temp count elements: %d", count);
+
     while (temp != NULL)
     {
         int min = find_min_in_array(temp->course_info.marks);