Add tests missing from initial commit