Added 100% code coverage tests