Set Variable Type Python Function Return