1 files changed, 10 insertions, 1 deletions
diff --git a/src/musichub/main/Main.java b/src/musichub/main/Main.java
index b72a95b..58080bd 100644
--- a/src/musichub/main/Main.java
+++ b/src/musichub/main/Main.java
@@ -2,11 +2,17 @@ package musichub.main;
 
 import musichub.business.*;
 
+import java.io.IOException;
+import java.util.ArrayList;
 import java.util.Iterator;
+import java.util.List;
 import java.util.Scanner;
 
+import javax.sound.sampled.LineUnavailableException;
+import javax.sound.sampled.UnsupportedAudioFileException;
+
 public class Main {
-    public static void main(String[] args) {
+    public static void main(String[] args) throws UnsupportedAudioFileException, IOException, LineUnavailableException {
 
         MusicHub theHub = new MusicHub();
 
@@ -55,7 +61,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.getSong(songs, song);
                     } catch (NoAlbumFoundException ex) {
                         System.out.println("No album found with the requested title " + ex.getMessage());
                     }