added more test. now 100% code coverage