Minor changes

Minor revisions to code