Adding missing tests
Adding missing tests