Add an example test

Ok. Updated as you suggested