diff --git a/kirjasto.py b/kirjasto.py
index 3bb5cc404c4c30d52bcead37517dee5513f0cb6a..efd809f58d8d0c1395a107d27e6883969d768b05 100644
--- a/kirjasto.py
+++ b/kirjasto.py
@@ -3,7 +3,6 @@ NIMI = 0
 KIRJAILIJA = 1
 VUOSI = 2
 SIVUJA = 3
-
 def lisaa_kirja(kirjat: list, nimi: str, kirjailija: str, vuosi: int, sivuja: int):
     """ Lisää annetun kirjan kirjalistaan """
     kirja = (nimi, kirjailija, vuosi, sivuja)
@@ -14,15 +13,15 @@ def etsi_kirja_nimen_perusteella(kirjat, hakusana: str) -> list:
     lista = []
     for kirja in kirjat:
         if hakusana in kirja[NIMI]:
-            return lista
-    
+            lista.append(kirja)
+    return lista
     
 
 def kirjailijan_kirjat(kirjat, kirjailija: str) -> list:
     """ Palauttaa kaikki annetun kirjailijan kirjoittamat kirjat """
     lista = []
     for kirja in kirjat:
-        if kirja[NIMI] == kirjailija:
+        if kirja[KIRJAILIJA] == kirjailija:
             lista.append(kirja)
 
     return lista
@@ -31,7 +30,7 @@ def vuoden_kirjat(kirjat, vuosi: int) -> list:
     """ Palauttaa kaikki kirjat, jotka on kirjoitettu annettuna vuonna """
     lista = []
     for kirja in kirjat:
-        if kirja[NIMI] == vuosi:
+        if kirja[VUOSI] == vuosi:
             lista.append(kirja)
 
     return lista
@@ -60,10 +59,9 @@ def tulosta_valikko():
 
 # PÄÄOHJELMA
 kirjat = []
-
+tulosta_valikko()
 while True:
-    tulosta_valikko()
-    valinta = input("Valinta: ")
+    valinta = int(input("Valinta: "))
     if valinta == 0:
         break
 
@@ -80,11 +78,11 @@ while True:
     elif valinta == 3:
         kirjailija = input("Anna kirjailja: ")
         lista = kirjailijan_kirjat(kirjat, kirjailija)
-        
+        tulosta_kirjat(lista)
     elif valinta == 4:
         vuosi = int(input("Anna vuosi: "))
         lista = vuoden_kirjat(kirjat, vuosi)
-        tulosta_kirjat(vuosi)
+        tulosta_kirjat(lista)
     elif valinta == 5:
         tulosta_kirjat(kirjat)