Updated with a working example