Add unit.py script to hold first unit tests