can you move those two lines after the `else`? It's not needed before.