Get example working
Add tests