- get unit tests working
