feat: add missing files
chore: update tests