Edit code format a bit