Also updated the example code so it's complete and consistent with other examples + removed unused import.