Added support for puppet 3.1.0