ps5.py final
Unit tests