Ended up adding the block of code from the article link you sent me :)