Making minor changes to code