Tests been added to reach 100% coverage