Function Visual Studio Code Python Getting Input