Added more unit tests for omod classes