assign_tseg function

Add segment numbers to observations