added two more unit tests for code that was not covered