Adding only a piece of better code.