Don't need the `done` before the `if/else`  statement