Address an edge case - input being null. Add test cases.