minor changes in code
bug fixed