added comments and fixed only bug

swaaaag