added comments/small fixes