Add code to run test, get malloc counts