Adding some initial tests

:eyes: this please @vkarthik26