Add new files to __init__