Small changes

Added more comments