updated comments in code