small code editing