impove code