Added some code to a few files.