slight change in code. minor corrections