Add linting and pytest