Adds new tests

Adds new tests