replace the duplciate code