Small update for style

Work unfinished. Ignore for now.