Added python2 check, should help resolve #51