Add missing files *blush*