Added check for python version