Clean up that code example just added