More changes suggested by referee