Add missing files from previous commit