Adding testing code.