Add include in logic