Added another JUnit test for better code coverage