[Issue #77][WIP] Added sub_model in test]