Add segment.io code