Add Python support