just updated code sample above