adding save class to submit