Added tests for the base module