Added more comments. Before submitting, remove TODO and maybe fix that comment