Add layout and reset module weight