Ah good point - I'll update with some changes :)