remove debug statement.
Add comment
Fix some pickling issues
Fix more pickling of FunctionGraph