Removing testing code