Unit Tests set up and working