code improvement 1