Hola:
Usando display, deja de estar el objeto, y el resto suele "re-ajustarse", así que en vez de display sea mejor visibility, que si bien sigue estando ahí, no estorba, y el contenido no se ajuste.
Sobre donde ponerlo, supongo que una línea antes del return.
Saludos
