Updated to integrate with colour thief