Finish unit testing