How to return values from python function