That's great to hear!
Unfortunately you will need to write it from scratch as the existing functionality is not open source. However, it is a pretty simple so you should be able to create your own version fairly easily.
What kind of extra functionality would you like to add? If you don't mind sharing your thoughts then we might be able to give you some design tips.