being a bit more lenient on linting. For now
Adding some more unit tests