Small code editing