Add helpers to feature tests