some small changes - putting some comments