Python print 1 or 2