updated comments for working version