Add Python environment setup