fixed error - added comments