Updated example and add test file