Update script to check python versions