Fixed bug, added comments