Python does not return a variable