1 files changed, 10 insertions, 2 deletions

diff --git a/src/musichub/main/Main.java b/src/main/java/musichub/main/Main.java
index b72a95b..c820b5d 100644
--- a/src/musichub/main/Main.java
+++ b/src/main/java/musichub/main/Main.java
@@ -2,11 +2,15 @@ package musichub.main;
 
 import musichub.business.*;
 
+import javax.sound.sampled.LineUnavailableException;
+import javax.sound.sampled.UnsupportedAudioFileException;
+import java.io.IOException;
 import java.util.Iterator;
+import java.util.List;
 import java.util.Scanner;
 
 public class Main {
-    public static void main(String[] args) {
+    public static void main(String[] args) throws UnsupportedAudioFileException, IOException, LineUnavailableException {
 
         MusicHub theHub = new MusicHub();
 
@@ -41,7 +45,8 @@ public class Main {
 
                     albumTitle = scan.nextLine();
                     try {
-                        System.out.println(theHub.getAlbumSongsSortedByGenre(albumTitle));
+                        List<Song> songs = theHub.getAlbumSongsSortedByGenre(albumTitle);
+                        System.out.println(songs);
                     } catch (NoAlbumFoundException ex) {
                         System.out.println("No album found with the requested title " + ex.getMessage());
                     }
@@ -55,7 +60,10 @@ public class Main {
 
                     albumTitle = scan.nextLine();
                     try {
+                        List<AudioElement> songs = theHub.getAlbumSongs(albumTitle);
                         System.out.println(theHub.getAlbumSongs(albumTitle));
+                        String song = scan.nextLine();
+                        theHub.getAudioElement(songs, song);
                     } catch (NoAlbumFoundException ex) {
                         System.out.println("No album found with the requested title " + ex.getMessage());
                     }