DRY up example a little