Added tests to increase coverage