fix: adding comments