Feature/test suite

Steve - please can you check my Python.