Fix compile issues (add missing files)