Adding basic testing framework

pep8 -- space before class