Update command to work with python2