make it work; add simple test