Python testing now(almost) working