pientä refaktorointia ja bugien korjailua