It worked!. I tried using save and restoring when i first reimplemented this class, but it wouldn't work as expected for some reason (dumb me probably)
Thanks a lot!