Add missing files and update tests