Add example and some tests

Add basic test and example