changed comments
warped it inside a class for better customization