Use pytest framework