Finally got Unit Tests set up