Add ifdef to include headers