diff --git a/notebooks/1-load-and-convert-statsbomb-data.ipynb b/notebooks/1-load-and-convert-statsbomb-data.ipynb
index f6ca5db..3eee4c0 100644
--- a/notebooks/1-load-and-convert-statsbomb-data.ipynb
+++ b/notebooks/1-load-and-convert-statsbomb-data.ipynb
@@ -155,6 +155,7 @@
        " 'La Liga',\n",
        " 'Liga Profesional',\n",
        " 'Ligue 1',\n",
+       " 'Major League Soccer',\n",
        " 'NWSL',\n",
        " 'North American League',\n",
        " 'Premier League',\n",
@@ -304,7 +305,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 9,
    "metadata": {
     "tags": []
    },
@@ -320,7 +321,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 306/306 [12:45<00:00,  2.50s/it]\n"
+      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 306/306 [10:19<00:00,  2.02s/it]\n"
      ]
     },
     {
@@ -334,7 +335,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [15:25<00:00,  2.44s/it]\n"
+      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [12:13<00:00,  1.93s/it]\n"
      ]
     },
     {
@@ -348,7 +349,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [15:33<00:00,  2.46s/it]\n"
+      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [12:04<00:00,  1.91s/it]\n"
      ]
     },
     {
@@ -362,7 +363,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 377/377 [15:46<00:00,  2.51s/it]\n"
+      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 377/377 [13:02<00:00,  2.08s/it]\n"
      ]
     },
     {
@@ -376,7 +377,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [15:54<00:00,  2.51s/it]\n"
+      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [14:30<00:00,  2.29s/it]\n"
      ]
     }
    ],
@@ -409,7 +410,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 10,
    "metadata": {},
    "outputs": [
     {
@@ -561,7 +562,7 @@
        "3890411           0  Olympiastadion Berlin    Peter Sippel  "
       ]
      },
-     "execution_count": 11,
+     "execution_count": 10,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -572,7 +573,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 11,
    "metadata": {},
    "outputs": [
     {
@@ -638,7 +639,7 @@
        "172                 Augsburg"
       ]
      },
-     "execution_count": 12,
+     "execution_count": 11,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -649,7 +650,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 12,
    "metadata": {},
    "outputs": [
     {
@@ -686,33 +687,33 @@
        "  </thead>\n",
        "  <tbody>\n",
        "    <tr>\n",
-       "      <th>20662</th>\n",
+       "      <th>3053</th>\n",
        "      <td>181</td>\n",
-       "      <td>Alexander Nübel</td>\n",
+       "      <td>Leroy Sané</td>\n",
        "      <td>NaN</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>37972</th>\n",
-       "      <td>175</td>\n",
-       "      <td>Jens Grahl</td>\n",
-       "      <td>NaN</td>\n",
+       "      <th>3499</th>\n",
+       "      <td>181</td>\n",
+       "      <td>Jean-Eric Maxim Choupo-Moting</td>\n",
+       "      <td>Eric Maxim Choupo-Moting</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>15146</th>\n",
-       "      <td>178</td>\n",
-       "      <td>Mike-Steven Bähre</td>\n",
-       "      <td>Mike Bähre</td>\n",
+       "      <th>3502</th>\n",
+       "      <td>181</td>\n",
+       "      <td>Joël Andre Job Matip</td>\n",
+       "      <td>Joël Matip</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>401018</th>\n",
-       "      <td>184</td>\n",
-       "      <td>Nico Rinderknecht</td>\n",
+       "      <th>3510</th>\n",
+       "      <td>181</td>\n",
+       "      <td>Sead Kolašinac</td>\n",
        "      <td>NaN</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>28601</th>\n",
-       "      <td>174</td>\n",
-       "      <td>Marvin Wanitzek</td>\n",
+       "      <th>5242</th>\n",
+       "      <td>181</td>\n",
+       "      <td>Younès Belhanda</td>\n",
        "      <td>NaN</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
@@ -720,16 +721,16 @@
        "</div>"
       ],
       "text/plain": [
-       "           team_id        player_name    nickname\n",
-       "player_id                                        \n",
-       "20662          181    Alexander Nübel         NaN\n",
-       "37972          175         Jens Grahl         NaN\n",
-       "15146          178  Mike-Steven Bähre  Mike Bähre\n",
-       "401018         184  Nico Rinderknecht         NaN\n",
-       "28601          174    Marvin Wanitzek         NaN"
+       "           team_id                    player_name                  nickname\n",
+       "player_id                                                                  \n",
+       "3053           181                     Leroy Sané                       NaN\n",
+       "3499           181  Jean-Eric Maxim Choupo-Moting  Eric Maxim Choupo-Moting\n",
+       "3502           181           Joël Andre Job Matip                Joël Matip\n",
+       "3510           181                 Sead Kolašinac                       NaN\n",
+       "5242           181                Younès Belhanda                       NaN"
       ]
      },
-     "execution_count": 13,
+     "execution_count": 12,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -740,7 +741,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": 13,
    "metadata": {},
    "outputs": [
     {
@@ -764,7 +765,7 @@
        "  <thead>\n",
        "    <tr style=\"text-align: right;\">\n",
        "      <th></th>\n",
-       "      <th></th>\n",
+       "      <th>game_id</th>\n",
        "      <th>period_id</th>\n",
        "      <th>team_id</th>\n",
        "      <th>player_id</th>\n",
@@ -774,7 +775,6 @@
        "      <th>timestamp</th>\n",
        "      <th>minute</th>\n",
        "      <th>second</th>\n",
-       "      <th>possession</th>\n",
        "      <th>...</th>\n",
        "      <th>team_name</th>\n",
        "      <th>duration</th>\n",
@@ -788,7 +788,6 @@
        "      <th>counterpress</th>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>game_id</th>\n",
        "      <th>event_id</th>\n",
        "      <th></th>\n",
        "      <th></th>\n",
@@ -815,18 +814,17 @@
        "  </thead>\n",
        "  <tbody>\n",
        "    <tr>\n",
-       "      <th rowspan=\"5\" valign=\"top\">3890561</th>\n",
        "      <th>41bd60ac-9b2c-4cb8-85aa-23ae71825c1e</th>\n",
+       "      <td>3890561</td>\n",
        "      <td>1</td>\n",
        "      <td>175</td>\n",
        "      <td>NaN</td>\n",
        "      <td>35</td>\n",
        "      <td>Starting XI</td>\n",
        "      <td>1</td>\n",
-       "      <td>1900-01-01 00:00:00.000</td>\n",
+       "      <td>0 days 00:00:00</td>\n",
        "      <td>0</td>\n",
        "      <td>0</td>\n",
-       "      <td>1</td>\n",
        "      <td>...</td>\n",
        "      <td>Hoffenheim</td>\n",
        "      <td>0.000000</td>\n",
@@ -841,16 +839,16 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>fbca533d-f3f4-4a86-b4a3-4fcae63592cf</th>\n",
+       "      <td>3890561</td>\n",
        "      <td>1</td>\n",
        "      <td>181</td>\n",
        "      <td>NaN</td>\n",
        "      <td>35</td>\n",
        "      <td>Starting XI</td>\n",
        "      <td>2</td>\n",
-       "      <td>1900-01-01 00:00:00.000</td>\n",
+       "      <td>0 days 00:00:00</td>\n",
        "      <td>0</td>\n",
        "      <td>0</td>\n",
-       "      <td>1</td>\n",
        "      <td>...</td>\n",
        "      <td>Schalke 04</td>\n",
        "      <td>0.000000</td>\n",
@@ -865,16 +863,16 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01</th>\n",
+       "      <td>3890561</td>\n",
        "      <td>1</td>\n",
        "      <td>175</td>\n",
        "      <td>NaN</td>\n",
        "      <td>18</td>\n",
        "      <td>Half Start</td>\n",
        "      <td>3</td>\n",
-       "      <td>1900-01-01 00:00:00.000</td>\n",
+       "      <td>0 days 00:00:00</td>\n",
        "      <td>0</td>\n",
        "      <td>0</td>\n",
-       "      <td>1</td>\n",
        "      <td>...</td>\n",
        "      <td>Hoffenheim</td>\n",
        "      <td>0.000000</td>\n",
@@ -889,16 +887,16 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>442128f8-2e38-491c-bf1e-b336e91757fa</th>\n",
+       "      <td>3890561</td>\n",
        "      <td>1</td>\n",
        "      <td>181</td>\n",
        "      <td>NaN</td>\n",
        "      <td>18</td>\n",
        "      <td>Half Start</td>\n",
        "      <td>4</td>\n",
-       "      <td>1900-01-01 00:00:00.000</td>\n",
+       "      <td>0 days 00:00:00</td>\n",
        "      <td>0</td>\n",
        "      <td>0</td>\n",
-       "      <td>1</td>\n",
        "      <td>...</td>\n",
        "      <td>Schalke 04</td>\n",
        "      <td>0.000000</td>\n",
@@ -913,16 +911,16 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>644e16d7-10ca-45f0-8128-fc0055d6f753</th>\n",
+       "      <td>3890561</td>\n",
        "      <td>1</td>\n",
        "      <td>175</td>\n",
        "      <td>8387.0</td>\n",
        "      <td>30</td>\n",
        "      <td>Pass</td>\n",
        "      <td>5</td>\n",
-       "      <td>1900-01-01 00:00:00.482</td>\n",
+       "      <td>0 days 00:00:00.482000</td>\n",
        "      <td>0</td>\n",
        "      <td>0</td>\n",
-       "      <td>2</td>\n",
        "      <td>...</td>\n",
        "      <td>Hoffenheim</td>\n",
        "      <td>0.453238</td>\n",
@@ -937,102 +935,86 @@
        "    </tr>\n",
        "  </tbody>\n",
        "</table>\n",
-       "<p>5 rows × 24 columns</p>\n",
+       "<p>5 rows × 25 columns</p>\n",
        "</div>"
       ],
       "text/plain": [
-       "                                              period_id  team_id  player_id  \\\n",
-       "game_id event_id                                                              \n",
-       "3890561 41bd60ac-9b2c-4cb8-85aa-23ae71825c1e          1      175        NaN   \n",
-       "        fbca533d-f3f4-4a86-b4a3-4fcae63592cf          1      181        NaN   \n",
-       "        b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01          1      175        NaN   \n",
-       "        442128f8-2e38-491c-bf1e-b336e91757fa          1      181        NaN   \n",
-       "        644e16d7-10ca-45f0-8128-fc0055d6f753          1      175     8387.0   \n",
-       "\n",
-       "                                              type_id    type_name  index  \\\n",
-       "game_id event_id                                                            \n",
-       "3890561 41bd60ac-9b2c-4cb8-85aa-23ae71825c1e       35  Starting XI      1   \n",
-       "        fbca533d-f3f4-4a86-b4a3-4fcae63592cf       35  Starting XI      2   \n",
-       "        b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01       18   Half Start      3   \n",
-       "        442128f8-2e38-491c-bf1e-b336e91757fa       18   Half Start      4   \n",
-       "        644e16d7-10ca-45f0-8128-fc0055d6f753       30         Pass      5   \n",
+       "                                      game_id  period_id  team_id  player_id  \\\n",
+       "event_id                                                                       \n",
+       "41bd60ac-9b2c-4cb8-85aa-23ae71825c1e  3890561          1      175        NaN   \n",
+       "fbca533d-f3f4-4a86-b4a3-4fcae63592cf  3890561          1      181        NaN   \n",
+       "b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01  3890561          1      175        NaN   \n",
+       "442128f8-2e38-491c-bf1e-b336e91757fa  3890561          1      181        NaN   \n",
+       "644e16d7-10ca-45f0-8128-fc0055d6f753  3890561          1      175     8387.0   \n",
        "\n",
-       "                                                           timestamp  minute  \\\n",
-       "game_id event_id                                                               \n",
-       "3890561 41bd60ac-9b2c-4cb8-85aa-23ae71825c1e 1900-01-01 00:00:00.000       0   \n",
-       "        fbca533d-f3f4-4a86-b4a3-4fcae63592cf 1900-01-01 00:00:00.000       0   \n",
-       "        b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01 1900-01-01 00:00:00.000       0   \n",
-       "        442128f8-2e38-491c-bf1e-b336e91757fa 1900-01-01 00:00:00.000       0   \n",
-       "        644e16d7-10ca-45f0-8128-fc0055d6f753 1900-01-01 00:00:00.482       0   \n",
+       "                                      type_id    type_name  index  \\\n",
+       "event_id                                                            \n",
+       "41bd60ac-9b2c-4cb8-85aa-23ae71825c1e       35  Starting XI      1   \n",
+       "fbca533d-f3f4-4a86-b4a3-4fcae63592cf       35  Starting XI      2   \n",
+       "b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01       18   Half Start      3   \n",
+       "442128f8-2e38-491c-bf1e-b336e91757fa       18   Half Start      4   \n",
+       "644e16d7-10ca-45f0-8128-fc0055d6f753       30         Pass      5   \n",
        "\n",
-       "                                              second  possession  ...  \\\n",
-       "game_id event_id                                                  ...   \n",
-       "3890561 41bd60ac-9b2c-4cb8-85aa-23ae71825c1e       0           1  ...   \n",
-       "        fbca533d-f3f4-4a86-b4a3-4fcae63592cf       0           1  ...   \n",
-       "        b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01       0           1  ...   \n",
-       "        442128f8-2e38-491c-bf1e-b336e91757fa       0           1  ...   \n",
-       "        644e16d7-10ca-45f0-8128-fc0055d6f753       0           2  ...   \n",
+       "                                                  timestamp  minute  second  \\\n",
+       "event_id                                                                      \n",
+       "41bd60ac-9b2c-4cb8-85aa-23ae71825c1e        0 days 00:00:00       0       0   \n",
+       "fbca533d-f3f4-4a86-b4a3-4fcae63592cf        0 days 00:00:00       0       0   \n",
+       "b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01        0 days 00:00:00       0       0   \n",
+       "442128f8-2e38-491c-bf1e-b336e91757fa        0 days 00:00:00       0       0   \n",
+       "644e16d7-10ca-45f0-8128-fc0055d6f753 0 days 00:00:00.482000       0       0   \n",
        "\n",
-       "                                               team_name  duration  \\\n",
-       "game_id event_id                                                     \n",
-       "3890561 41bd60ac-9b2c-4cb8-85aa-23ae71825c1e  Hoffenheim  0.000000   \n",
-       "        fbca533d-f3f4-4a86-b4a3-4fcae63592cf  Schalke 04  0.000000   \n",
-       "        b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01  Hoffenheim  0.000000   \n",
-       "        442128f8-2e38-491c-bf1e-b336e91757fa  Schalke 04  0.000000   \n",
-       "        644e16d7-10ca-45f0-8128-fc0055d6f753  Hoffenheim  0.453238   \n",
+       "                                      ...   team_name  duration  \\\n",
+       "event_id                              ...                         \n",
+       "41bd60ac-9b2c-4cb8-85aa-23ae71825c1e  ...  Hoffenheim  0.000000   \n",
+       "fbca533d-f3f4-4a86-b4a3-4fcae63592cf  ...  Schalke 04  0.000000   \n",
+       "b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01  ...  Hoffenheim  0.000000   \n",
+       "442128f8-2e38-491c-bf1e-b336e91757fa  ...  Schalke 04  0.000000   \n",
+       "644e16d7-10ca-45f0-8128-fc0055d6f753  ...  Hoffenheim  0.453238   \n",
        "\n",
-       "                                                                                          extra  \\\n",
-       "game_id event_id                                                                                  \n",
-       "3890561 41bd60ac-9b2c-4cb8-85aa-23ae71825c1e  {'tactics': {'formation': 3421, 'lineup': [{'p...   \n",
-       "        fbca533d-f3f4-4a86-b4a3-4fcae63592cf  {'tactics': {'formation': 4141, 'lineup': [{'p...   \n",
-       "        b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01                                                 {}   \n",
-       "        442128f8-2e38-491c-bf1e-b336e91757fa                                                 {}   \n",
-       "        644e16d7-10ca-45f0-8128-fc0055d6f753  {'pass': {'recipient': {'id': 5460, 'name': 'A...   \n",
+       "                                                                                  extra  \\\n",
+       "event_id                                                                                  \n",
+       "41bd60ac-9b2c-4cb8-85aa-23ae71825c1e  {'tactics': {'formation': 3421, 'lineup': [{'p...   \n",
+       "fbca533d-f3f4-4a86-b4a3-4fcae63592cf  {'tactics': {'formation': 4141, 'lineup': [{'p...   \n",
+       "b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01                                                 {}   \n",
+       "442128f8-2e38-491c-bf1e-b336e91757fa                                                 {}   \n",
+       "644e16d7-10ca-45f0-8128-fc0055d6f753  {'pass': {'recipient': {'id': 5460, 'name': 'A...   \n",
        "\n",
-       "                                                                      related_events  \\\n",
-       "game_id event_id                                                                       \n",
-       "3890561 41bd60ac-9b2c-4cb8-85aa-23ae71825c1e                                      []   \n",
-       "        fbca533d-f3f4-4a86-b4a3-4fcae63592cf                                      []   \n",
-       "        b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01  [442128f8-2e38-491c-bf1e-b336e91757fa]   \n",
-       "        442128f8-2e38-491c-bf1e-b336e91757fa  [b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01]   \n",
-       "        644e16d7-10ca-45f0-8128-fc0055d6f753  [7602c8d9-d988-4eae-bb9f-309fbad4c7c5]   \n",
+       "                                                              related_events  \\\n",
+       "event_id                                                                       \n",
+       "41bd60ac-9b2c-4cb8-85aa-23ae71825c1e                                      []   \n",
+       "fbca533d-f3f4-4a86-b4a3-4fcae63592cf                                      []   \n",
+       "b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01  [442128f8-2e38-491c-bf1e-b336e91757fa]   \n",
+       "442128f8-2e38-491c-bf1e-b336e91757fa  [b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01]   \n",
+       "644e16d7-10ca-45f0-8128-fc0055d6f753  [7602c8d9-d988-4eae-bb9f-309fbad4c7c5]   \n",
        "\n",
-       "                                             player_name  position_id  \\\n",
-       "game_id event_id                                                        \n",
-       "3890561 41bd60ac-9b2c-4cb8-85aa-23ae71825c1e         NaN          NaN   \n",
-       "        fbca533d-f3f4-4a86-b4a3-4fcae63592cf         NaN          NaN   \n",
-       "        b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01         NaN          NaN   \n",
-       "        442128f8-2e38-491c-bf1e-b336e91757fa         NaN          NaN   \n",
-       "        644e16d7-10ca-45f0-8128-fc0055d6f753    Mark Uth         18.0   \n",
+       "                                     player_name position_id  \\\n",
+       "event_id                                                       \n",
+       "41bd60ac-9b2c-4cb8-85aa-23ae71825c1e         NaN         NaN   \n",
+       "fbca533d-f3f4-4a86-b4a3-4fcae63592cf         NaN         NaN   \n",
+       "b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01         NaN         NaN   \n",
+       "442128f8-2e38-491c-bf1e-b336e91757fa         NaN         NaN   \n",
+       "644e16d7-10ca-45f0-8128-fc0055d6f753    Mark Uth        18.0   \n",
        "\n",
-       "                                                         position_name  \\\n",
-       "game_id event_id                                                         \n",
-       "3890561 41bd60ac-9b2c-4cb8-85aa-23ae71825c1e                       NaN   \n",
-       "        fbca533d-f3f4-4a86-b4a3-4fcae63592cf                       NaN   \n",
-       "        b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01                       NaN   \n",
-       "        442128f8-2e38-491c-bf1e-b336e91757fa                       NaN   \n",
-       "        644e16d7-10ca-45f0-8128-fc0055d6f753  Right Attacking Midfield   \n",
+       "                                                 position_name      location  \\\n",
+       "event_id                                                                       \n",
+       "41bd60ac-9b2c-4cb8-85aa-23ae71825c1e                       NaN           NaN   \n",
+       "fbca533d-f3f4-4a86-b4a3-4fcae63592cf                       NaN           NaN   \n",
+       "b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01                       NaN           NaN   \n",
+       "442128f8-2e38-491c-bf1e-b336e91757fa                       NaN           NaN   \n",
+       "644e16d7-10ca-45f0-8128-fc0055d6f753  Right Attacking Midfield  [61.0, 40.1]   \n",
        "\n",
-       "                                                  location under_pressure  \\\n",
-       "game_id event_id                                                            \n",
-       "3890561 41bd60ac-9b2c-4cb8-85aa-23ae71825c1e           NaN          False   \n",
-       "        fbca533d-f3f4-4a86-b4a3-4fcae63592cf           NaN          False   \n",
-       "        b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01           NaN          False   \n",
-       "        442128f8-2e38-491c-bf1e-b336e91757fa           NaN          False   \n",
-       "        644e16d7-10ca-45f0-8128-fc0055d6f753  [61.0, 40.1]          False   \n",
+       "                                     under_pressure counterpress  \n",
+       "event_id                                                          \n",
+       "41bd60ac-9b2c-4cb8-85aa-23ae71825c1e          False        False  \n",
+       "fbca533d-f3f4-4a86-b4a3-4fcae63592cf          False        False  \n",
+       "b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01          False        False  \n",
+       "442128f8-2e38-491c-bf1e-b336e91757fa          False        False  \n",
+       "644e16d7-10ca-45f0-8128-fc0055d6f753          False        False  \n",
        "\n",
-       "                                              counterpress  \n",
-       "game_id event_id                                            \n",
-       "3890561 41bd60ac-9b2c-4cb8-85aa-23ae71825c1e         False  \n",
-       "        fbca533d-f3f4-4a86-b4a3-4fcae63592cf         False  \n",
-       "        b15ba6b1-61ac-4d9c-b2a3-096ce31bcf01         False  \n",
-       "        442128f8-2e38-491c-bf1e-b336e91757fa         False  \n",
-       "        644e16d7-10ca-45f0-8128-fc0055d6f753         False  \n",
-       "\n",
-       "[5 rows x 24 columns]"
+       "[5 rows x 25 columns]"
       ]
      },
-     "execution_count": 15,
+     "execution_count": 13,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -1043,7 +1025,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 16,
+   "execution_count": 14,
    "metadata": {},
    "outputs": [
     {
@@ -1067,7 +1049,7 @@
        "  <thead>\n",
        "    <tr style=\"text-align: right;\">\n",
        "      <th></th>\n",
-       "      <th></th>\n",
+       "      <th>game_id</th>\n",
        "      <th>original_event_id</th>\n",
        "      <th>period_id</th>\n",
        "      <th>time_seconds</th>\n",
@@ -1082,7 +1064,6 @@
        "      <th>bodypart_id</th>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>game_id</th>\n",
        "      <th>action_id</th>\n",
        "      <th></th>\n",
        "      <th></th>\n",
@@ -1096,15 +1077,16 @@
        "      <th></th>\n",
        "      <th></th>\n",
        "      <th></th>\n",
+       "      <th></th>\n",
        "    </tr>\n",
        "  </thead>\n",
        "  <tbody>\n",
        "    <tr>\n",
-       "      <th rowspan=\"5\" valign=\"top\">3890561</th>\n",
        "      <th>0</th>\n",
+       "      <td>3890561</td>\n",
        "      <td>644e16d7-10ca-45f0-8128-fc0055d6f753</td>\n",
        "      <td>1</td>\n",
-       "      <td>0.0</td>\n",
+       "      <td>0.482</td>\n",
        "      <td>175</td>\n",
        "      <td>8387.0</td>\n",
        "      <td>53.33125</td>\n",
@@ -1117,9 +1099,10 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>1</th>\n",
+       "      <td>3890561</td>\n",
        "      <td>329a1879-2521-4614-8c68-b4798b0e5d23</td>\n",
        "      <td>1</td>\n",
-       "      <td>0.0</td>\n",
+       "      <td>0.935</td>\n",
        "      <td>175</td>\n",
        "      <td>5460.0</td>\n",
        "      <td>52.63125</td>\n",
@@ -1132,9 +1115,10 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>2</th>\n",
+       "      <td>3890561</td>\n",
        "      <td>77e2ddaf-6de3-49e7-a318-7d765799b543</td>\n",
        "      <td>1</td>\n",
-       "      <td>1.0</td>\n",
+       "      <td>1.015</td>\n",
        "      <td>175</td>\n",
        "      <td>5460.0</td>\n",
        "      <td>51.93125</td>\n",
@@ -1147,9 +1131,10 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>3</th>\n",
+       "      <td>3890561</td>\n",
        "      <td>1b91a029-f722-4b0d-b9d5-53cdc776f9e3</td>\n",
        "      <td>1</td>\n",
-       "      <td>2.0</td>\n",
+       "      <td>2.167</td>\n",
        "      <td>175</td>\n",
        "      <td>6039.0</td>\n",
        "      <td>47.11875</td>\n",
@@ -1162,9 +1147,10 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>4</th>\n",
+       "      <td>3890561</td>\n",
        "      <td>2c51f271-c812-45af-896b-06f49a14a5bb</td>\n",
        "      <td>1</td>\n",
-       "      <td>2.0</td>\n",
+       "      <td>2.954</td>\n",
        "      <td>175</td>\n",
        "      <td>6039.0</td>\n",
        "      <td>45.71875</td>\n",
@@ -1180,32 +1166,32 @@
        "</div>"
       ],
       "text/plain": [
-       "                                      original_event_id  period_id  \\\n",
-       "game_id action_id                                                    \n",
-       "3890561 0          644e16d7-10ca-45f0-8128-fc0055d6f753          1   \n",
-       "        1          329a1879-2521-4614-8c68-b4798b0e5d23          1   \n",
-       "        2          77e2ddaf-6de3-49e7-a318-7d765799b543          1   \n",
-       "        3          1b91a029-f722-4b0d-b9d5-53cdc776f9e3          1   \n",
-       "        4          2c51f271-c812-45af-896b-06f49a14a5bb          1   \n",
+       "           game_id                     original_event_id  period_id  \\\n",
+       "action_id                                                             \n",
+       "0          3890561  644e16d7-10ca-45f0-8128-fc0055d6f753          1   \n",
+       "1          3890561  329a1879-2521-4614-8c68-b4798b0e5d23          1   \n",
+       "2          3890561  77e2ddaf-6de3-49e7-a318-7d765799b543          1   \n",
+       "3          3890561  1b91a029-f722-4b0d-b9d5-53cdc776f9e3          1   \n",
+       "4          3890561  2c51f271-c812-45af-896b-06f49a14a5bb          1   \n",
        "\n",
-       "                   time_seconds  team_id  player_id   start_x  start_y  \\\n",
-       "game_id action_id                                                        \n",
-       "3890561 0                   0.0      175     8387.0  53.33125  33.9575   \n",
-       "        1                   0.0      175     5460.0  52.63125  35.8275   \n",
-       "        2                   1.0      175     5460.0  51.93125  35.4875   \n",
-       "        3                   2.0      175     6039.0  47.11875  32.2575   \n",
-       "        4                   2.0      175     6039.0  45.71875  29.6225   \n",
+       "           time_seconds  team_id  player_id   start_x  start_y     end_x  \\\n",
+       "action_id                                                                  \n",
+       "0                 0.482      175     8387.0  53.33125  33.9575  52.63125   \n",
+       "1                 0.935      175     5460.0  52.63125  35.8275  51.93125   \n",
+       "2                 1.015      175     5460.0  51.93125  35.4875  47.11875   \n",
+       "3                 2.167      175     6039.0  47.11875  32.2575  45.71875   \n",
+       "4                 2.954      175     6039.0  45.71875  29.6225  29.96875   \n",
        "\n",
-       "                      end_x    end_y  type_id  result_id  bodypart_id  \n",
-       "game_id action_id                                                      \n",
-       "3890561 0          52.63125  35.8275        0          1            4  \n",
-       "        1          51.93125  35.4875       21          1            0  \n",
-       "        2          47.11875  32.2575        0          1            4  \n",
-       "        3          45.71875  29.6225       21          1            0  \n",
-       "        4          29.96875  15.3425        0          1            5  "
+       "             end_y  type_id  result_id  bodypart_id  \n",
+       "action_id                                            \n",
+       "0          35.8275        0          1            4  \n",
+       "1          35.4875       21          1            0  \n",
+       "2          32.2575        0          1            4  \n",
+       "3          29.6225       21          1            0  \n",
+       "4          15.3425        0          1            5  "
       ]
      },
-     "execution_count": 16,
+     "execution_count": 14,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -1216,7 +1202,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 17,
+   "execution_count": 15,
    "metadata": {
     "tags": []
    },
@@ -1224,18 +1210,11 @@
    "source": [
     "dataset.close()"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "/home/pieterr/Jupiter/Projects/soccer_xg",
+   "display_name": "soccer_xg",
    "language": "python",
    "name": "soccer_xg"
   },
diff --git a/notebooks/1-load-and-convert-wyscout-data.ipynb b/notebooks/1-load-and-convert-wyscout-data.ipynb
index daaaf9f..0e7b2b2 100644
--- a/notebooks/1-load-and-convert-wyscout-data.ipynb
+++ b/notebooks/1-load-and-convert-wyscout-data.ipynb
@@ -291,21 +291,21 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Loading game data...:  86%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▌                             | 327/380 [05:01<00:50,  1.06it/s]/cw/dtaijupiter/NoCsBack/dtai/pieterr/Projects/soccer_xg/.venv/lib/python3.11/site-packages/socceraction/data/wyscout/loader.py:281: UserWarning: A player with ID=0 was substituted in the 90th minute of game 2576016, but could not be found on the bench.\n",
+      "Loading game data...:  86%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▋                                          | 327/380 [03:54<00:41,  1.29it/s]/cw/dtaijupiter/NoCsBack/dtai/pieterr/Projects/soccer_xg/.venv/lib/python3.11/site-packages/socceraction/data/wyscout/loader.py:281: UserWarning: A player with ID=0 was substituted in the 90th minute of game 2576016, but could not be found on the bench.\n",
       "  warnings.warn(\n",
-      "Loading game data...:  98%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████    | 373/380 [05:42<00:05,  1.20it/s]/cw/dtaijupiter/NoCsBack/dtai/pieterr/Projects/soccer_xg/.venv/lib/python3.11/site-packages/socceraction/data/wyscout/loader.py:281: UserWarning: A player with ID=0 was substituted in the 56th minute of game 2575965, but could not be found on the bench.\n",
+      "Loading game data...:  98%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▍     | 373/380 [04:25<00:04,  1.59it/s]/cw/dtaijupiter/NoCsBack/dtai/pieterr/Projects/soccer_xg/.venv/lib/python3.11/site-packages/socceraction/data/wyscout/loader.py:281: UserWarning: A player with ID=0 was substituted in the 56th minute of game 2575965, but could not be found on the bench.\n",
       "  warnings.warn(\n",
       "/cw/dtaijupiter/NoCsBack/dtai/pieterr/Projects/soccer_xg/.venv/lib/python3.11/site-packages/socceraction/data/wyscout/loader.py:281: UserWarning: A player with ID=0 was substituted in the 62th minute of game 2575965, but could not be found on the bench.\n",
       "  warnings.warn(\n",
       "/cw/dtaijupiter/NoCsBack/dtai/pieterr/Projects/soccer_xg/.venv/lib/python3.11/site-packages/socceraction/data/wyscout/loader.py:281: UserWarning: A player with ID=0 was substituted in the 88th minute of game 2575965, but could not be found on the bench.\n",
       "  warnings.warn(\n",
-      "Loading game data...:  99%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▎ | 377/380 [05:46<00:02,  1.03it/s]/cw/dtaijupiter/NoCsBack/dtai/pieterr/Projects/soccer_xg/.venv/lib/python3.11/site-packages/socceraction/data/wyscout/loader.py:281: UserWarning: A player with ID=0 was substituted in the 74th minute of game 2575959, but could not be found on the bench.\n",
+      "Loading game data...:  99%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▌  | 377/380 [04:28<00:02,  1.31it/s]/cw/dtaijupiter/NoCsBack/dtai/pieterr/Projects/soccer_xg/.venv/lib/python3.11/site-packages/socceraction/data/wyscout/loader.py:281: UserWarning: A player with ID=0 was substituted in the 74th minute of game 2575959, but could not be found on the bench.\n",
       "  warnings.warn(\n",
       "/cw/dtaijupiter/NoCsBack/dtai/pieterr/Projects/soccer_xg/.venv/lib/python3.11/site-packages/socceraction/data/wyscout/loader.py:281: UserWarning: A player with ID=0 was substituted in the 81th minute of game 2575959, but could not be found on the bench.\n",
       "  warnings.warn(\n",
       "/cw/dtaijupiter/NoCsBack/dtai/pieterr/Projects/soccer_xg/.venv/lib/python3.11/site-packages/socceraction/data/wyscout/loader.py:281: UserWarning: A player with ID=0 was substituted in the 84th minute of game 2575959, but could not be found on the bench.\n",
       "  warnings.warn(\n",
-      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [05:48<00:00,  1.09it/s]\n"
+      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [04:29<00:00,  1.41it/s]\n"
      ]
     },
     {
@@ -319,7 +319,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [05:41<00:00,  1.11it/s]\n"
+      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [04:22<00:00,  1.44it/s]\n"
      ]
     },
     {
@@ -333,7 +333,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [05:47<00:00,  1.09it/s]\n"
+      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [04:30<00:00,  1.41it/s]\n"
      ]
     },
     {
@@ -347,7 +347,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [05:47<00:00,  1.09it/s]\n"
+      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 380/380 [04:21<00:00,  1.45it/s]\n"
      ]
     },
     {
@@ -361,7 +361,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 306/306 [04:41<00:00,  1.09it/s]\n"
+      "Loading game data...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 306/306 [03:26<00:00,  1.48it/s]\n"
      ]
     }
    ],
@@ -372,7 +372,7 @@
     "    mode=\"w\"\n",
     ")\n",
     "for comp in comps:\n",
-    "    # map wyscout IDs to internal IDs\n",
+    "    # get name and id of competition\n",
     "    competition_name, competition_id = comp['league']['name'], comp['league']['wy_id']\n",
     "    season_name, season_id = comp['season']['name'], comp['season']['wy_id']\n",
     "    print(f\"Importing {competition_name} {season_name} ...\")\n",
@@ -635,47 +635,47 @@
        "  </thead>\n",
        "  <tbody>\n",
        "    <tr>\n",
-       "      <th>99430</th>\n",
-       "      <td>3158</td>\n",
-       "      <td>Łukasz Skorupski</td>\n",
-       "      <td>Ł. Skorupski</td>\n",
+       "      <th>21384</th>\n",
+       "      <td>3162</td>\n",
+       "      <td>Ciro Immobile</td>\n",
+       "      <td>C. Immobile</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>22053</th>\n",
-       "      <td>3172</td>\n",
-       "      <td>Francesco Rossi</td>\n",
-       "      <td>F. Rossi</td>\n",
+       "      <th>20550</th>\n",
+       "      <td>3162</td>\n",
+       "      <td>Ştefan Daniel Radu</td>\n",
+       "      <td>Ş. Radu</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>485464</th>\n",
-       "      <td>3219</td>\n",
-       "      <td>Oliva Christian Rutjens</td>\n",
-       "      <td>Christian Rutjens</td>\n",
+       "      <th>130</th>\n",
+       "      <td>3162</td>\n",
+       "      <td>Stefan de Vrij</td>\n",
+       "      <td>S. de Vrij</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>354470</th>\n",
-       "      <td>3157</td>\n",
-       "      <td>Emanuele Torrasi</td>\n",
-       "      <td>E. Torrasi</td>\n",
+       "      <th>346908</th>\n",
+       "      <td>3162</td>\n",
+       "      <td>Alessandro Murgia</td>\n",
+       "      <td>A. Murgia</td>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>343915</th>\n",
-       "      <td>3204</td>\n",
-       "      <td>Gabriele Marchegiani</td>\n",
-       "      <td>G. Marchegiani</td>\n",
+       "      <th>376362</th>\n",
+       "      <td>3162</td>\n",
+       "      <td>Luiz Felipe Ramos Marchi</td>\n",
+       "      <td>Luiz Felipe</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
        "</table>\n",
        "</div>"
       ],
       "text/plain": [
-       "           team_id              player_name           nickname\n",
-       "player_id                                                     \n",
-       "99430         3158         Łukasz Skorupski       Ł. Skorupski\n",
-       "22053         3172          Francesco Rossi           F. Rossi\n",
-       "485464        3219  Oliva Christian Rutjens  Christian Rutjens\n",
-       "354470        3157         Emanuele Torrasi         E. Torrasi\n",
-       "343915        3204     Gabriele Marchegiani     G. Marchegiani"
+       "           team_id               player_name     nickname\n",
+       "player_id                                                \n",
+       "21384         3162             Ciro Immobile  C. Immobile\n",
+       "20550         3162        Ştefan Daniel Radu      Ş. Radu\n",
+       "130           3162            Stefan de Vrij   S. de Vrij\n",
+       "346908        3162         Alessandro Murgia    A. Murgia\n",
+       "376362        3162  Luiz Felipe Ramos Marchi  Luiz Felipe"
       ]
      },
      "execution_count": 11,
@@ -690,7 +690,9 @@
   {
    "cell_type": "code",
    "execution_count": 12,
-   "metadata": {},
+   "metadata": {
+    "tags": []
+   },
    "outputs": [
     {
      "data": {
@@ -713,7 +715,170 @@
        "  <thead>\n",
        "    <tr style=\"text-align: right;\">\n",
        "      <th></th>\n",
+       "      <th>game_id</th>\n",
+       "      <th>period_id</th>\n",
+       "      <th>milliseconds</th>\n",
+       "      <th>team_id</th>\n",
+       "      <th>player_id</th>\n",
+       "      <th>type_id</th>\n",
+       "      <th>type_name</th>\n",
+       "      <th>subtype_id</th>\n",
+       "      <th>subtype_name</th>\n",
+       "      <th>positions</th>\n",
+       "      <th>tags</th>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>event_id</th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>253668302</th>\n",
+       "      <td>2576335</td>\n",
+       "      <td>1</td>\n",
+       "      <td>2417.590</td>\n",
+       "      <td>3161</td>\n",
+       "      <td>3344</td>\n",
+       "      <td>8</td>\n",
+       "      <td>Pass</td>\n",
+       "      <td>85</td>\n",
+       "      <td>Simple pass</td>\n",
+       "      <td>[{'y': 50, 'x': 49}, {'y': 58, 'x': 38}]</td>\n",
+       "      <td>[{'id': 1801}]</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>253668303</th>\n",
+       "      <td>2576335</td>\n",
+       "      <td>1</td>\n",
+       "      <td>3904.412</td>\n",
+       "      <td>3161</td>\n",
+       "      <td>116349</td>\n",
+       "      <td>8</td>\n",
+       "      <td>Pass</td>\n",
+       "      <td>85</td>\n",
+       "      <td>Simple pass</td>\n",
+       "      <td>[{'y': 58, 'x': 38}, {'y': 91, 'x': 37}]</td>\n",
+       "      <td>[{'id': 1801}]</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>253668304</th>\n",
+       "      <td>2576335</td>\n",
+       "      <td>1</td>\n",
+       "      <td>6484.211</td>\n",
+       "      <td>3161</td>\n",
+       "      <td>135903</td>\n",
+       "      <td>8</td>\n",
+       "      <td>Pass</td>\n",
+       "      <td>85</td>\n",
+       "      <td>Simple pass</td>\n",
+       "      <td>[{'y': 91, 'x': 37}, {'y': 72, 'x': 34}]</td>\n",
+       "      <td>[{'id': 1801}]</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>253668306</th>\n",
+       "      <td>2576335</td>\n",
+       "      <td>1</td>\n",
+       "      <td>10043.835</td>\n",
+       "      <td>3161</td>\n",
+       "      <td>138408</td>\n",
+       "      <td>8</td>\n",
+       "      <td>Pass</td>\n",
+       "      <td>85</td>\n",
+       "      <td>Simple pass</td>\n",
+       "      <td>[{'y': 72, 'x': 34}, {'y': 14, 'x': 36}]</td>\n",
+       "      <td>[{'id': 1801}]</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>253668308</th>\n",
+       "      <td>2576335</td>\n",
+       "      <td>1</td>\n",
+       "      <td>14032.070</td>\n",
+       "      <td>3161</td>\n",
+       "      <td>21094</td>\n",
+       "      <td>8</td>\n",
+       "      <td>Pass</td>\n",
+       "      <td>85</td>\n",
+       "      <td>Simple pass</td>\n",
+       "      <td>[{'y': 14, 'x': 36}, {'y': 39, 'x': 30}]</td>\n",
+       "      <td>[{'id': 1801}]</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "           game_id  period_id  milliseconds  team_id  player_id  type_id  \\\n",
+       "event_id                                                                   \n",
+       "253668302  2576335          1      2417.590     3161       3344        8   \n",
+       "253668303  2576335          1      3904.412     3161     116349        8   \n",
+       "253668304  2576335          1      6484.211     3161     135903        8   \n",
+       "253668306  2576335          1     10043.835     3161     138408        8   \n",
+       "253668308  2576335          1     14032.070     3161      21094        8   \n",
+       "\n",
+       "          type_name  subtype_id subtype_name  \\\n",
+       "event_id                                       \n",
+       "253668302      Pass          85  Simple pass   \n",
+       "253668303      Pass          85  Simple pass   \n",
+       "253668304      Pass          85  Simple pass   \n",
+       "253668306      Pass          85  Simple pass   \n",
+       "253668308      Pass          85  Simple pass   \n",
+       "\n",
+       "                                          positions            tags  \n",
+       "event_id                                                             \n",
+       "253668302  [{'y': 50, 'x': 49}, {'y': 58, 'x': 38}]  [{'id': 1801}]  \n",
+       "253668303  [{'y': 58, 'x': 38}, {'y': 91, 'x': 37}]  [{'id': 1801}]  \n",
+       "253668304  [{'y': 91, 'x': 37}, {'y': 72, 'x': 34}]  [{'id': 1801}]  \n",
+       "253668306  [{'y': 72, 'x': 34}, {'y': 14, 'x': 36}]  [{'id': 1801}]  \n",
+       "253668308  [{'y': 14, 'x': 36}, {'y': 39, 'x': 30}]  [{'id': 1801}]  "
+      ]
+     },
+     "execution_count": 12,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "dataset.events(game_id=2576335).head()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 13,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
        "      <th></th>\n",
+       "      <th>game_id</th>\n",
        "      <th>period_id</th>\n",
        "      <th>time_seconds</th>\n",
        "      <th>team_id</th>\n",
@@ -728,7 +893,6 @@
        "      <th>result_id</th>\n",
        "    </tr>\n",
        "    <tr>\n",
-       "      <th>game_id</th>\n",
        "      <th>action_id</th>\n",
        "      <th></th>\n",
        "      <th></th>\n",
@@ -742,12 +906,13 @@
        "      <th></th>\n",
        "      <th></th>\n",
        "      <th></th>\n",
+       "      <th></th>\n",
        "    </tr>\n",
        "  </thead>\n",
        "  <tbody>\n",
        "    <tr>\n",
-       "      <th rowspan=\"5\" valign=\"top\">2576335</th>\n",
        "      <th>0</th>\n",
+       "      <td>2576335</td>\n",
        "      <td>1</td>\n",
        "      <td>2.417590</td>\n",
        "      <td>3161</td>\n",
@@ -763,6 +928,7 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>1</th>\n",
+       "      <td>2576335</td>\n",
        "      <td>1</td>\n",
        "      <td>3.904412</td>\n",
        "      <td>3161</td>\n",
@@ -778,6 +944,7 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>2</th>\n",
+       "      <td>2576335</td>\n",
        "      <td>1</td>\n",
        "      <td>6.484211</td>\n",
        "      <td>3161</td>\n",
@@ -793,6 +960,7 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>3</th>\n",
+       "      <td>2576335</td>\n",
        "      <td>1</td>\n",
        "      <td>10.043835</td>\n",
        "      <td>3161</td>\n",
@@ -808,6 +976,7 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>4</th>\n",
+       "      <td>2576335</td>\n",
        "      <td>1</td>\n",
        "      <td>14.032070</td>\n",
        "      <td>3161</td>\n",
@@ -826,32 +995,32 @@
        "</div>"
       ],
       "text/plain": [
-       "                   period_id  time_seconds  team_id  player_id  start_x  \\\n",
-       "game_id action_id                                                         \n",
-       "2576335 0                  1      2.417590     3161       3344    53.55   \n",
-       "        1                  1      3.904412     3161     116349    65.10   \n",
-       "        2                  1      6.484211     3161     135903    66.15   \n",
-       "        3                  1     10.043835     3161     138408    69.30   \n",
-       "        4                  1     14.032070     3161      21094    67.20   \n",
+       "           game_id  period_id  time_seconds  team_id  player_id  start_x  \\\n",
+       "action_id                                                                  \n",
+       "0          2576335          1      2.417590     3161       3344    53.55   \n",
+       "1          2576335          1      3.904412     3161     116349    65.10   \n",
+       "2          2576335          1      6.484211     3161     135903    66.15   \n",
+       "3          2576335          1     10.043835     3161     138408    69.30   \n",
+       "4          2576335          1     14.032070     3161      21094    67.20   \n",
        "\n",
-       "                   start_y  end_x  end_y original_event_id  bodypart_id  \\\n",
-       "game_id action_id                                                         \n",
-       "2576335 0            34.00  65.10  39.44         253668302            0   \n",
-       "        1            39.44  66.15  61.88         253668303            0   \n",
-       "        2            61.88  69.30  48.96         253668304            0   \n",
-       "        3            48.96  67.20   9.52         253668306            0   \n",
-       "        4             9.52  73.50  26.52         253668308            0   \n",
+       "           start_y  end_x  end_y original_event_id  bodypart_id  type_id  \\\n",
+       "action_id                                                                  \n",
+       "0            34.00  65.10  39.44         253668302            0        0   \n",
+       "1            39.44  66.15  61.88         253668303            0        0   \n",
+       "2            61.88  69.30  48.96         253668304            0        0   \n",
+       "3            48.96  67.20   9.52         253668306            0        0   \n",
+       "4             9.52  73.50  26.52         253668308            0        0   \n",
        "\n",
-       "                   type_id  result_id  \n",
-       "game_id action_id                      \n",
-       "2576335 0                0          1  \n",
-       "        1                0          1  \n",
-       "        2                0          1  \n",
-       "        3                0          1  \n",
-       "        4                0          1  "
+       "           result_id  \n",
+       "action_id             \n",
+       "0                  1  \n",
+       "1                  1  \n",
+       "2                  1  \n",
+       "3                  1  \n",
+       "4                  1  "
       ]
      },
-     "execution_count": 12,
+     "execution_count": 13,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -862,7 +1031,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 14,
    "metadata": {
     "tags": []
    },
@@ -870,18 +1039,11 @@
    "source": [
     "dataset.close()"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "/home/pieterr/Jupiter/Projects/soccer_xg",
+   "display_name": "soccer_xg",
    "language": "python",
    "name": "soccer_xg"
   },