diff --git a/src/main/java/models/Course.java b/src/main/java/models/Course.java index 17d5b3b..2901c3b 100644 --- a/src/main/java/models/Course.java +++ b/src/main/java/models/Course.java @@ -34,17 +34,18 @@ public class Course { } public List getStudents() { - return students; + return students; } - public void showStudents() { + public String getStudentsDisplay() { if (students.isEmpty()) { - System.out.println(" No students enrolled yet."); + return " No students enrolled yet."; } else { - System.out.println(" Students in " + name + ":"); + StringBuilder sb = new StringBuilder(" Students in " + name + ":\n"); for (Student s : students) { - System.out.println(" - " + s.getUsername()); + sb.append(" - ").append(s.getUsername()).append("\n"); } + return sb.toString(); } }