Made it a mod now - added framework