Updated new tests slightly