Fixed some output, changes usage from unittest to pytest