Add test.py with api calls