added comment to code and improved readme