make changes suggested in review