Add functionality test cases