Adding behat functionality