Add some init code