Small fixes(added comment)