Adapt main.py to entry of test to numpy