I can look into that, will need some consideration for more robust customization. If you want you can share some ideas here or in github issues.