Added two new tests