Undo not working for actions done through scripts as buttons

Hi all,

I am new to scripting in max and I'm already enjoying this. you guys are doing great job here, I am learning so much from you all.. :)

This is my first Question here! I have created few scripts and using them as button in a customized toolbar. But the actions performed through them, like move the object to Origin, is undoable. The action is being done but its not getting registered in the history so Max is not getting what to undo. Is there any solution for this? Thank you in advance :)