Python version check added