Add extension check