Adding unit tests to MLB