Add support for running just unit tests