Add in minor changes