small code changes.