Add Python loading code