improve the code