Fixed DI file and added pro