Add improvement to code