Adding unit tests and changing the framework for async setups