Add missing files
Fix #4