Add check for python