Fixing odd styling in code