changed simple case to import function instead of use IO files
changed source problem to import python function directly