I think this is possible with some refactoring, I will work on adding it.