Added missing tests

added basic tests