Cover all functionality with test cases
Cover all functionality with test cases