Add tests for MainModule