got unit testing more or less working