Adding testing functionality.