Add include; add test