add more testing code