Adding testing code