imporve code