Разобрался. Нужно не через print, а через return данные выводить :)