linting
Adding tests