Added test file
Added main.py