Add relations imports when option is chosen