Added tests for integrated code in module