Tein koodista vähän hienompaa ja lisäsin testejä.