Minor changes in code and code documention