Added tests.py, part of #6