Added missing files for build