desired behavior

align(..., {
  ...
})

current behavior :

align(
  ...,
  {
    ...
  }
)

The idea is, if args excluding the last one can fit in a single line, then the last arg is not indented. Otherwise the bahavior is same as original