Update the extension for Behat 3 RC

This reimplements the extension for Behat 3 RC