Add missed code files