Tom Schwartz Is Focusing on 'Levity' After 'Floundering' From Katie Maloney Divorce