Fix and add testing for categorize_by_function.py