I made some small code changes.