Add missing files

Fix #3