Documentation added and style for code added