Updating simple net to save traces

Adding initial testing framework support