Just checking in here. I've added a bit of functionality for what [I think] should happen.