modify the dimension, check main function