changed the tests.

Code is now tested and working on a pi