Add missing `#include` statements