Add some fixes
Add example test