Feature/tests fix

Fixed tests by removing `tests.py`: `run_tests` script uses `demo` to run tests