Made a few tweaks to the code in the repository.