Yeah that would work as well, I'm going to fix it one way or another when the voting period is over :) Thanks for the idea!
