Add level, revise weight map

Add level, revise weight map