Made a few little code changes.