Add new extension class to capture screen shots on failure