Improved save method to accept args and kwargs