Round out the example so it's fully functional