refs #328 Fixed code documentation
refs #328 - Small code refactoring