Forgot to add new file custom.py for custom testing sets.