improve IG
fix bug, improve code
improve IG