modify main.py with simple program
added executable line and 'import numpy'