Add more zimplish implementation of TMR problem