few modifications in code