Add module tests