updating example code and working on a test