Why move this into the main file?