adding comments and testing setup