Some changes in code && added minor documentation