adding more testing code