hieman parannusta koodiin