I was able to change the code a bit to make this possible.