Extends pytest functionality
