I added a variable for this, see updated code :).