Small changes, added comment, add variable