improving of code