Adding new tests to increase coverage