small main change