That should be doable pretty easily. The code is easy to read and modify.