Added Pytest run code