added comment, updated code