Added initial tests
Added initial tests