While not the most romantic activity, drafting a prenup is smart. It means you, not a court, define rules for asset division if you divorce. That can include equity comp.