Think I've implemented it!!!