Interactive input for python