add nice unit test