fix comments
Fixed some issues, now working :-)!