You're right! Would've come to that when testing extending the prototype version. For now will mark this WIP