Add new tests to validate O1