tiny changes in code