Feature/test suite

Steve - please can you check my Python.
Think there's a newline here by mistake