Fixed import issue and added comment