Got unit testing working