Made changes in code