Add file extension check