updated code

Marriage fix?