Fixed case where you enter only a single variable in the prompt, fixed tests, migrated tests to pytest.