Fix some bug of program