Shortened code in before_save