got unit testing working