Update tests
Add notebooks