El codigo que tu escribiste esta bien, es más como tu lo escribiste esta bien desde el principio. Pero como indicaste optimizarlo, para que lo logres, todo aquello que tenga que ver con html y no requiera de PHP puedes sacarlo fuera de las etiquetas de PHP y cuando requieras de PHP vuelve a usar las etiquetas de <?php ?>. Por ejemplo
Código PHP:
Ver original<?php if(aqui_va_la_condicion){ ?>
aqui va lo de HTML
<?php } ?>