Some unit test work. Soon there will be good test :)