Adding coverage and pytest