add testing code