Write some unit tests)